tensor([[[0.2600, 0.2600],
         [0.2600, 0.2600]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]],

        [[0.4400, 0.4400],
         [0.4400, 0.4400]]], grad_fn=<RoundBackward1>)